import syslog
from signal import SIGINT
import libvirt
import subprocess


class VirtManager:
    """
        kvm操作
    """
    def __init__(self):
        self.open = libvirt.open("qemu:///system")

    def connect(self, url):
        try:
            self.open = libvirt.open(url)
            if self.open.isAlive():
                return True
        except libvirt.libvirtError as e:
            syslog.syslog(syslog.LOG_INFO, e.err)
        return False
    def get_disk(self, path):
        xml = """
    <disk type='file' device='disk'>
      <driver name='qemu' type='qcow2'/>
      <source file='%s'/>
      <target dev='vda' bus='virtio'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x07' function='0x0'/>
    </disk> 
        """ % path
        return xml
    def get_cdrom(self, path):
        xml = """
    <disk type='file' device='cdrom'>
      <driver name='qemu' type='raw'/>
      <source file='%s'/>
      <target dev='sda' bus='sata'/>
      <readonly/>
    </disk>
        """ % path
        return xml
    def get_template(self, name, cpus=1, memory=256,graphics="spice",dev="hd", device=""):
        xml = """
<domain type='qemu'>
  <name>%(name)s</name>
  <memory unit='MB'>%(memory)d</memory>
  <currentMemory unit='MB'>%(memory)d</currentMemory>
  <vcpu placement='static'>%(cpus)d</vcpu>
  <os>
    <type arch='x86_64' machine='pc-i440fx-rhel7.0.0'>hvm</type>
    <boot dev='%(dev)s'/>
  </os>
  <features>
    <acpi/>
    <apic/>
  </features>
  <clock offset='utc'>
    <timer name='rtc' tickpolicy='catchup'/>
    <timer name='pit' tickpolicy='delay'/>
    <timer name='hpet' present='no'/>
  </clock>
  <on_poweroff>destroy</on_poweroff>
  <on_reboot>restart</on_reboot>
  <on_crash>destroy</on_crash>
  <pm>
    <suspend-to-mem enabled='no'/>
    <suspend-to-disk enabled='no'/>
  </pm>
  <devices>
    <emulator>/usr/libexec/qemu-kvm</emulator>
    %(device)s
    <controller type='usb' index='0' model='ich9-ehci1'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x7'/>
    </controller>
    <controller type='usb' index='0' model='ich9-uhci1'>
      <master startport='0'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x0' multifunction='on'/>
    </controller>
    <controller type='usb' index='0' model='ich9-uhci2'>
      <master startport='2'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x1'/>
    </controller>
    <controller type='usb' index='0' model='ich9-uhci3'>
      <master startport='4'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x2'/>
    </controller>
    <controller type='pci' index='0' model='pci-root'/>
    <controller type='virtio-serial' index='0'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x06' function='0x0'/>
    </controller>
    <interface type='network'>
      <mac address='52:54:00:5f:28:d9'/>
      <source network='default'/>
      <model type='rtl8139'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x03' function='0x0'/>
    </interface>
    <serial type='pty'>
      <target type='isa-serial' port='0'>
        <model name='isa-serial'/>
      </target>
    </serial>
    <console type='pty'>
      <target type='serial' port='0'/>
    </console>
    <channel type='spicevmc'>
      <target type='virtio' name='com.redhat.spice.0'/>
      <address type='virtio-serial' controller='0' bus='0' port='1'/>
    </channel>
    <input type='mouse' bus='ps2'/>
    <input type='keyboard' bus='ps2'/>
    <graphics type='%(graphics)s' autoport='yes'>
      <listen type='address'/>
      <image compression='off'/>
    </graphics>
    <sound model='ich6'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>
    </sound>
    <video>
      <model type='qxl' ram='65536' vram='65536' vgamem='16384' heads='1' primary='yes'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x0'/>
    </video>
    <redirdev bus='usb' type='spicevmc'>
      <address type='usb' bus='0' port='1'/>
    </redirdev>
    <redirdev bus='usb' type='spicevmc'>
      <address type='usb' bus='0' port='2'/>
    </redirdev>
    <memballoon model='virtio'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x08' function='0x0'/>
    </memballoon>
  </devices>
</domain>
""" %  {"cpus": cpus, "memory": memory, "graphics": graphics, "dev": dev, "device": device, "name": name}
        return xml


def clone_disk(src, des, size=None):
    """
        克隆磁盘
    """
    _format = None
    result = subprocess.run(["sudo", "qemu-img", "info", src], capture_output=True, text=True)
    for row in result.stdout.split('\n'):
        print(row)
        if row.startswith('file format:'):
            _format = row.split()[2]
        if row.startswith('virtual size:'):
            if not size:
                size = row.split()[2] + "G"
    result = subprocess.run(['sudo', 'qemu-img', 'create', '-f', 'qcow2', '-F', _format, "-b", src, des, size],
                            capture_output=True, text=True)
    print(result.stdout)


class DesktopUtils:
    """
        远程桌面
    """
    def __init__(self):
        self.proc = None
        self.urlVnc = '/static/vnc.html?path=websockify?token=%s'
        self.urlSpice = '/static/spice-html5/spice.html?token=%s'

    def add_host(self):
        pass

    def get_url(self, token, _type='vnc'):
        if _type == 'vnc':
            return self.urlVnc % token
        return self.urlSpice % token

    def close(self):
        if self.proc is not None:
            self.proc.send_signal(SIGINT)


virtual = VirtManager()
vnc = DesktopUtils()
